package cn.tedu.v5;

import java.io.*;
import java.net.*;
import java.util.*;

/**
 * Version 5 实现多人聊天+QQ聊天+消息转发
 * 功能: 【登录 + 聊天 + 退出】
 * 登录步骤: 1.私有化属性，保存账号密码---userDataBase
 *          2.Server构造器，创建两个用户名和密码
 *          3.处理多个客户端，加私有化属性: 账号和密码
 */
public class Server {
    private ServerSocket serverSocket;
    private static final Integer PORT=6666;
    private List<PrintWriter> allOut=new ArrayList<>();
    // 1.私有化属性，保存账号密码---userDataBase
    private Map<String,String> userDataBase;

    public Server(){
        try {
            serverSocket=new ServerSocket(PORT);
           // 2.Server对象启动--main方法---创建服务器之初，完成对象账号和密码赋值
            userDataBase=new HashMap<>();
            userDataBase.put("空桑少主","123456");
            userDataBase.put("鹄羹","654321");
            userDataBase.put("玻璃鱿鱼","123");
        } catch (BindException e){
            System.out.println("端口号重复申请，建议换个端口号");
        } catch (IOException e) {
            System.out.println("申请端口号失败...");
        }
    }

    public void start()  {
        try {
            while (true){
            System.out.println("-------等待客户端链接------");
            Socket socket = serverSocket.accept();

            new Thread(new ClientHandler(socket)).start();
            }
        } catch (IOException e) {
            System.out.println("服务器端通讯断开");
        } finally {
            if (serverSocket!=null&&!serverSocket.isClosed()){
                try {
                    serverSocket.close();
                } catch (IOException e) {
                    System.out.println("服务器关闭失败");
                }
            }
        }
    }

    public static void main(String[] args)  {
        new Server().start();
    }

    //此处未调用broadcastMessage转发消息的方法，方法颜色是灰色
    //需要调用broadcastMessage方法才能实现"消息广播"功能
    private synchronized void broadcastMessage(String message){
        for (PrintWriter pw:allOut){
            pw.println("消息广播"+message);
        }
    }

    private class ClientHandler implements Runnable{
        private Socket socket;
        private String ip;
        private BufferedReader br;
        private PrintWriter pw;
        // 3.很多客户端具备账号密码
        private String username;
        private String password;

        public ClientHandler(Socket s){
            this.socket=s;
            ip=socket.getInetAddress().getHostAddress();

            try {
                //流不可以折行，只有字符串可以折行
                //字符串折行可以拼接，但流不可以
                 br=new BufferedReader(new InputStreamReader(socket.getInputStream(),"UTF8"));
                 pw = new PrintWriter(new BufferedWriter(new OutputStreamWriter(socket.getOutputStream(),"UTF8")), true);

                synchronized (allOut) {
                    allOut.add(pw);
                }
                //不可以在此处关闭，客户端会出现报空情况
                /*for (PrintWriter pw :allOut.toArray(new PrintWriter[0])){
                    pw.close();
                    break;
                }*/
            } catch (IOException e) {
                System.out.println("客户端断开通讯~");
            }
        }

        //4.4.1 封装判断账号与密码是否正确的方法---Login
        private boolean Login(String user,String pass){
            //判断Map中是否存在这个账号和密码
            //userDataBase.containsKey(user) 判断账号存在
            //userDataBase.get(user).equals(pass) 判断这个账号密码是否正确
            return userDataBase.containsKey(user)&&userDataBase.get(user).equals(pass);
        }

        //5.账号与密码验证成功---登录--聊天:读写功能---封装方法handlerChat()
        private void handlerChat(){
            new Thread(()->{
                try{
                    String message;
                    while ((message=br.readLine())!=null){
                        System.out.println(username+"客户端说: "+message);
                        System.out.println("服务器端请输入: ");
                    //在服务器端读到客户端的消息，调用：broadcastMessage方法转发给其他客户端

                        //在服务器端就可以判断如果输入是exit，则客户端断开链接
                        if(message.equalsIgnoreCase("exit")){
                            pw.println(username+"退出聊天室！");
                             break; //这个是跳出程序，并不是结束程序，所以只能用break
                            //而服务器端不能断开，因为还有其他客户端链接
                        }
                        broadcastMessage(message);
                        //服务器端向客户端推送退出功能提示:
                        pw.println("请输入消息 (输入 exit 退出):");
                    }
                }catch (IOException e){
                    System.out.println("客户端读入失败!");
                }
            }).start();

            Scanner input=new Scanner(System.in);
            while(true){
                System.out.println(ip+"服务器请输入: ");
                String say=input.nextLine();
             //   pw.println("服务器说: "+say);   输出了两次服务器端说
                //聊天室退出功能的实现: 验证服务器是否发送给客户端exit,服务端跳出程序
                if (say.equalsIgnoreCase("exit")){
                    System.out.println("服务器端即将关闭...");
                   // System.exit(0);
                    break;//跳出程序，服务器端不能直接结束程序，否则其他客户端无法链接服务器
                }
                pw.println("服务器端说: "+say);
            }
        }
        public void run() {
            try {
            // 4.目的: 启动客户端与服务器端链接
            // 假设:一个客户端链接上了，服务器怎么处理
            //4.1 登录验证  服务器端和客户端的交互问题
            pw.println("请输入用户名: ");  //发给客户端
            username=br.readLine();  //读入客户端发来的账号
            //4.2 密码验证
            pw.println("请输入密码: ");  //发给客户端
            password=br.readLine();  //读入客户端发来的账号
            //4.3 判断账号密码是否正确---账号很多的情况
            //4.4 账号很多情况: 封装方法---功能: 判断账号与密码是否正确
            if (Login(username,password)){  //判断Login 为true  账号密码正确
                pw.println("登录成功，欢迎进入聊天室");
            //聊天  4.5 封装一个方法---登录成功--开始聊天: 读写功能--HandlerChat
                handlerChat();
            }else {  //不通过
                pw.println("登录失败，用户名或者密码错误!");
                socket.close();  //登录失败，关闭链接
            }
            } catch (NullPointerException e) {
                System.out.println("IO流传输对象，有对象为null,请立马核对。"
                        +"建议: 查看try块中对象可能赋值为null");
            } catch (IOException e){
                System.out.println("服务器端断开连接");
            }finally {
                synchronized (allOut) {
                    allOut.remove(pw);
                }
                //聊天室退出功能的实现: 2.关闭写出流---不可以在此处关闭
             /*   for (PrintWriter pw :allOut.toArray(new PrintWriter[0])){
                    pw.close();
                    break;
                }*/
               /* System.exit(0);*/
                if (br!=null) {
                    try {
                        br.close();
                    } catch (IOException e) {
                        System.out.println("通讯关闭失败~");
                    }
                }
                if (pw!=null) pw.close();
                if (socket!=null&&!socket.isClosed()){
                    try {
                        socket.close();
                    } catch (IOException e) {
                        System.out.println("通讯关闭失败~");
                    }
                }
            }
        }
    }
}
